#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# @Author: renjin@bit.edu.cn
# @Date  : 2024-08-07

from spirecv.dataloader.ImgDirReaderNode import ImgDirReaderNode
from spirecv.algorithm.common_det.YOLOv5v7DetNode import YOLOv5v7DetNode
from spirecv.saver.SpireViewSaverNode import SpireViewSaverNode


job_name = 'detection'
parameter_file = '../../params/spirecv2/default_params.json'


dataloader = ImgDirReaderNode(job_name, param_dict_or_file=parameter_file)
algorithm = YOLOv5v7DetNode(job_name, param_dict_or_file=parameter_file)
saver = SpireViewSaverNode(job_name, param_dict_or_file=parameter_file)

dataloader.launch_next()

